import chromadb

client = chromadb.HttpClient(host='chromadb', port=8000)

collection = client.get_or_create_collection(name='key_collection', metadata={"hnsw:space": "cosine"})

collection.add(
	ids=['1', '2'],
	embeddings=[[1., 1.], [2., 2.]],
)

print(collection.get('1', include=['embeddings']))
